import MySQLdb

# 第一步：创建连接对象
from sql_week1.partial__01 import create_connect, random_pass

conn = create_connect(database='hrs')
# host主机地址
try:
    # 第二部：通过链接对象获取游标对象
    with conn.cursor() as cursor:

        # 第三步：通过游标对象向数据库服务器发出SQL语句
        for i in range(1, 1001):
            username, password = f'user{i}', random_pass(length=8)
            cursor.execute(
                    'insert into `tb_user`(`user_name`,`user_pass`)'
                    'values(%s, %s)',
                    (username, password)
            )

    # 第四步：通过连接对象提交事务
    conn.commit()
except MySQLdb.MySQLError as err:
    print(err)
    # 第四步：通过连接对象回滚事务
    conn.rollback()

finally:
    # 第五步：关闭连接
    # 数据库可以链接的游标数是有限的用完了之后要及时释放连接
    conn.close()

